#!/usr/bin/env bash

DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"

export PYTHONPATH=${DIR}/../../:${PYTHONPATH}
export PYTHONPATH=${DIR}/../../slim:${PYTHONPATH}
export PYTHONPATH=${DIR}/../:${PYTHONPATH}

# ssd

#TRAIN=/home/admins/cmake/ssd_mobilenet_v1_coco_11_06_2017
#
#python3 ${DIR}/../train.py \
#--train_dir ${TRAIN} \
#--pipeline_config_path ${TRAIN}/ssd_mobilenet_beer.config

# faster-rcnn

TRAIN=/home/admins/cmake/beer-faster

python3 ${DIR}/../train.py \
--train_dir ${TRAIN} \
--pipeline_config_path ${TRAIN}/faster_rcnn_inception_v2_beer.config

DATA=/home/admins/data/beer_data
POSTFIX=patch

python3 ${DIR}/train.py \
--data_dir ${DATA} \
--model_dir ${DATA}/classify \
--batch_size 64 \
--resnet_size 18 \
--train_number 58124 \
--train_epochs 200